package hg.demo.mybatis.plus.config;

import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import org.springframework.context.annotation.Bean;

/**
 * mybatisplus-实体bean生成器配置
 * @author hgSuper
 * @date 2020-07-06
 */
public class MyBatisPlusGeneratorConfig {

    /**
     * 初始化生成器
     * @return
     */
    public AutoGenerator createAutoGenerator() {
        AutoGenerator autoGenerator = new AutoGenerator();
        return autoGenerator;
    }

    /**
     * 配置生成路径
     * @return
     */
    @Bean
    public com.baomidou.mybatisplus.generator.config.GlobalConfig initGlobalConfig() {
        com.baomidou.mybatisplus.generator.config.GlobalConfig gc = new com.baomidou.mybatisplus.generator.config.GlobalConfig();
        String projectPath = System.getProperty("user.dir");
        gc.setOutputDir(projectPath + "/src/main/java");
        gc.setAuthor("hgSuper");
        gc.setOpen(false);
        gc.setOpen(false);
        return gc;
    }

    /**
     * 配置代码包
     * @return
     */
    @Bean
    public PackageConfig initPackageConfig() {
        PackageConfig pc = new PackageConfig();
        // 包名
        pc.setModuleName("");
        // 父目录下创建包
        pc.setParent("com.wy");
        return pc;
    }
}
